package com.pqj.pqaiagent.tool;

import org.jsoup.Jsoup;
import org.jsoup.nodes.Document;
import org.springframework.ai.tool.annotation.Tool;
import org.springframework.ai.tool.annotation.ToolParam;

/**
 * 网页抓取工具
 */
public class WebScrapingTool {

    @Tool(description = "Scrape the content of a web page") // 标记为工具方法，描述为"抓取网页内容"
    public String scrapeWebPage(@ToolParam(description = "URL of the web page to scrape") String url) {  // 参数：要抓取的网页URL
        try {
            // 使用Jsoup连接并获取网页内容
            Document document = Jsoup.connect(url).get();// 建立HTTP连接并获取HTML文档
            return document.html(); // 返回整个HTML文档的字符串表示
        } catch (Exception e) {
            return "Error scraping web page: " + e.getMessage();
        }
    }
}
